//
//  Problem11.swift
//  LeetCode - 11 - 盛最多水的容器
//
//  Created by 武侠 on 2020/7/8.
//  Copyright © 2020 zhulongiMac02. All rights reserved.
//

import UIKit
/*
 [11 盛最多水的容器]
 给你 n 个非负整数 a1，a2，...，an，每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线，垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线，使得它们与 x 轴共同构成的容器可以容纳最多的水。

 说明：你不能倾斜容器，且 n 的值至少为 2。

 示例：
 输入：[1,8,6,2,5,4,8,3,7]
 输出：49
 */

class Problem11: NSObject {
    func solution() {
        print("n = ", maxArea2([1,8,6,2,5,4,8,3,7]))
    }
    
    // 简单的暴力法:
    func maxArea(_ height: [Int]) -> Int {
        var st:Int = 0
        for i in 0..<height.count {
            for j in i+1..<height.count {
                var temp: Int = 0
                if height[i] < height[j] {
                    temp = (j - i) * height[i]
                } else {
                    temp = (j - i) * height[j]
                }
                if temp > st {
                    st = temp
                }
                print(i, j, height[i], height[j], temp)
            }
        }
        return st
    }
    
    // 双指针解法:
    // 思路: 设置i在0的位置, j在n-1的位置,
    // 但height[i] < height[j]: 这时要考虑移动i,还是j,
    // 假如:移动j, 那么 height[i] * (j - i) >= height[i] * (j - 1 - i)
    // 所以要移动 小的指i针,
    func maxArea2(_ height: [Int]) -> Int {
        var i: Int = 0
        var j: Int = height.count - 1
        var st: Int = 0
        
        while i < j {
            var temp: Int = 0
            if height[i] < height[j] {
                temp = height[i] * (j - i)
                i += 1
            } else if height[i] > height[j] {
                temp = height[j] * (j - i)
                j -= 1
            } else {
                temp = height[i] * (j - i)
                i += 1
                j -= 1
            }
            if temp > st {
                st = temp
            }
        }
        return st;
    }
}
